import os
import shutil


def organize_files(input_folder, output_folder, batch_size=100):
	# 创建输出文件夹，如果不存在
	if not os.path.exists(output_folder):
		os.makedirs(output_folder)
	
	# 获取所有文件，并按顺序排序
	files = sorted(os.listdir(input_folder))
	
	# 初始化文件计数器和文件夹编号
	file_count = 0
	folder_count = 1
	
	# 创建第一个子文件夹
	current_folder = os.path.join(output_folder, f"batch_{folder_count}")
	os.makedirs(current_folder)
	
	for file in files:
		# 只处理以.txt结尾的文件
		if file.endswith('.txt'):
			# 移动文件到当前子文件夹
			shutil.move(os.path.join(input_folder, file), current_folder)
			file_count += 1
			
			# 每100个文件切换到下一个文件夹
			if file_count == batch_size:
				folder_count += 1
				current_folder = os.path.join(output_folder, f"batch_{folder_count}")
				os.makedirs(current_folder)
				file_count = 0
	
	print("文件整理完成！")


if __name__ == "__main__":
	# 当前文件夹
	current_directory = os.getcwd()
	input_folder = os.path.join(current_directory, 'output')
	output_folder = os.path.join(current_directory, 'organized_output')
	
	organize_files(input_folder, output_folder)
